package com.webserver.http;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.net.Socket;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;

public class HttpServletResponse {
    private Socket socket;
    private int statusCode = 200;
    private String statusReason = "OK";
    private File contentFile;
    private Map<String, String> headers = new HashMap<>();

    public HttpServletResponse(Socket socket) {
        this.socket = socket;
    }

    public void response() throws IOException {

        Thread thread =Thread.currentThread();
        System.out.println("666666666666666666666666当前线程为："+thread+"6666666666666666666666666");

        sendStatusLine();
        sendHeaders();
        sendContent();
    }

    private void sendStatusLine() throws IOException {
        println("HTTP/1.1" + statusCode + statusReason);
    }

    private void sendHeaders() throws IOException {
        Set<Map.Entry<String, String>> entrySet = headers.entrySet();
        for (Map.Entry<String, String> e : entrySet) {
            String key = e.getKey();
            String value = e.getValue();
            println(key+": "+value); // addHeader(key, value);///////////////
        }
           println("");            //////////////////////////////////////////////
}

    private void addHeader(String key, String value) {
        headers.put(key, value);

    }

    private void sendContent() throws IOException {
       if (contentFile!=null){
           OutputStream out = socket.getOutputStream();
           FileInputStream fis = new FileInputStream(contentFile);
           int len;
           byte[] data = new byte[10 * 1024];
           while ((len = fis.read(data)) != -1) {
               out.write(data, 0, len);
           }
           System.out.println("-------------已经输出结果---------");
       }
    }

    private void println(String line) throws IOException {
        OutputStream out = socket.getOutputStream();
        byte[] data = line.getBytes(StandardCharsets.ISO_8859_1);
        out.write(data);
        out.write(13);
        out.write(10);

    }

    public File getContentFile() {
        return contentFile;
    }

    public void setContentFile(File contentFile) {
        this.contentFile = contentFile;
        String contentType = null;
        try {
            contentType = Files.probeContentType(contentFile.toPath());
        } catch (IOException e) {
            e.printStackTrace();
        }
        addHeader("Content-Type", contentType);
        addHeader("Content-Length",contentFile.length()+"");

    }

    public int getStatusCode() {
        return statusCode;
    }

    public void setStatusCode(int statusCode) {
        this.statusCode = statusCode;
    }

    public String getStatusReason() {
        return statusReason;
    }

    public void setStatusReason(String statusReason) {
        this.statusReason = statusReason;
    }
//    public void setHeaders(Map<String,String> headers){
//        this.headers=headers;
//    }

}
